from fastapi import FastAPI, Request
import ollama






if __name__ == '__main__':
    user_input = '帮我分析A行情'
    # 这里可以修改请求，比如添加系统提示词
    messages = [
        {"role": "system", "content": "你是一个专业的金融助手，只提供严谨的金融分析。"},
        {"role": "user", "content": user_input}
    ]


    response = ollama.chat(model='deepseek-r1:8b', messages=messages)
    print(response['message']['content'])

